package com.beidasoft.web.util;

import com.alibaba.fastjson.JSONObject;
import io.minio.*;
import io.minio.errors.*;
import io.minio.messages.Bucket;
import io.minio.messages.DeleteObject;
import io.minio.messages.Item;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.time.ZonedDateTime;
import java.util.*;

@Component
@Slf4j
public class MinioUtils {
    //private static final Logger logger = LoggerFactory.getLogger(this.getClass());
    private static final String SEPARATOR = "/";
    public static String endpoint;
    public static String bucketName;
    public static String accessKey;
    public static String secretKey;
    private static MinioClient minioClient;

    private MinioUtils() {
    }
    public static void init(String endpoint, String bucketName, String accessKey, String secretKey) {
        String borderStr = "|-------------------------------------------------------------------------------------------|";
        System.out.println(borderStr);
        System.out.println("                            endpoint：" + endpoint);
        System.out.println("                            bucketName：" + bucketName);
        System.out.println("                            accessKey：" + accessKey);
        System.out.println(borderStr);

        MinioUtils.endpoint = endpoint;
        MinioUtils.bucketName = bucketName;
        MinioUtils.accessKey = accessKey;
        MinioUtils.secretKey = secretKey;
        createMinioClient();
    }
    /**
     * 创建minioClient
     */
    public static void createMinioClient() {
        try {
            if (null == minioClient) {
                minioClient = MinioClient.builder().endpoint(endpoint).credentials(accessKey, secretKey).build();
                createBucket();
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 获取上传文件的基础路径
     *
     * @return url
     */
    public static String getBasisUrl() {
        return endpoint + SEPARATOR + bucketName + SEPARATOR;
    }


    /**
     * 初始化Bucket
     *
     * @throws Exception 异常
     */
    private static void createBucket() {
        try {
            if (!minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build())) {
                minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }

    }

    /**
     * 验证bucketName是否存在
     *
     * @return boolean true:存在
     */
    public static boolean bucketExists()
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
    }

    /**
     * 创建bucket
     *
     * @param bucketName bucket名称
     */
    public static void createBucket(String bucketName)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException, RegionConflictException {
        if (!minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build())) {
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
        }
    }

    /**
     * 获取全部bucket
     * <p>
     * https://docs.minio.io/cn/java-client-api-reference.html#listBuckets
     */
    public static List<Bucket> getAllBuckets()
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient.listBuckets();
    }

    /**
     * 判断文件是否存在
     *
     * @param bucketName 存储桶
     * @param objectName 对象
     * @return true：存在
     */
    //20240425 已做迁移
    public static boolean doesObjectExist(String bucketName, String objectName) {
        boolean exist = true;
        try {
            minioClient
                    .statObject(StatObjectArgs.builder().bucket(bucketName).object(objectName).build());
        } catch (Exception e) {
            exist = false;
        }
        return exist;
    }

    /**
     * 判断文件夹是否存在
     *
     * @param bucketName 存储桶
     * @param objectName 文件夹名称（去掉/）
     * @return true：存在
     */
    //20240424 已经做迁移   ks3服务器在判断文件是否存在的基础上，增加 /
    public static boolean doesFolderExist(String bucketName, String objectName) {
        boolean exist = false;
        try {
            Iterable<Result<Item>> results = minioClient.listObjects(
                    ListObjectsArgs.builder().bucket(bucketName).prefix(objectName).recursive(false).build());
            for (Result<Item> result : results) {
                Item item = result.get();
                String name = URLDecoder.decode(item.objectName(), "UTF-8");
                if (item.isDir() && objectName.equals(name.substring(0,name.length()-1))) {
                    exist = true;
                }
            }
        } catch (Exception e) {
            exist = false;
        }
        return exist;
    }


    /**
     * 获取文件流
     *
     * @param bucketName bucket名称
     * @param objectName 文件名称
     * @return 二进制流
     */
    //20240423已经迁移
    public static InputStream getObject(String bucketName, String objectName)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient
                .getObject(GetObjectArgs.builder().bucket(bucketName).object(objectName).build());
    }

    /**
     * 获取路径下文件列表
     *
     * @param bucketName bucket名称
     * @param prefix     文件名称
     * @param recursive  是否递归查找，如果是false,就模拟文件夹结构查找
     * @return 二进制流
     */
    public static Iterable<Result<Item>> listObjects(String bucketName, String prefix,
                                                     boolean recursive) {
        return minioClient.listObjects(
                ListObjectsArgs.builder().bucket(bucketName).prefix(prefix).recursive(recursive).build());
    }

    /**
     * 通过MultipartFile，上传文件
     *      由于minion私有桶最大天数为7点，所以把私用桶改为了public，直接访问返回地址可以访问
     * @param bucketName 存储桶
     * @param file       文件
     * @param objectName 对象名
     */
    //已经做迁移
    public static String putObject(String bucketName, MultipartFile file,
                                   String objectName, String contentType)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        InputStream inputStream = file.getInputStream();
        ObjectWriteResponse objectWriteResponse = minioClient.putObject(
                PutObjectArgs.builder().bucket(bucketName).object(objectName).contentType(contentType)
                        .stream(
                                inputStream, inputStream.available(), -1)
                        .build());

        String bucket = objectWriteResponse.bucket();
        String object = objectWriteResponse.object();

        return endpoint + "/" + bucket + "/" + object;
    }

    /**
     * 上传本地文件
     *
     * @param bucketName 存储桶
     * @param objectName 对象名称(文件实际存放地址)
     * @param fileName   本地文件路径
     */
    //已经做迁移
    public static ObjectWriteResponse putObject(String bucketName, String objectName,
                                                String fileName)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient.uploadObject(
                UploadObjectArgs.builder()
                        .bucket(bucketName).object(objectName).filename(fileName).build());
    }

    /**
     * 通过流上传文件
     *
     * @param bucketName  存储桶
     * @param objectName  文件对象
     * @param inputStream 文件流
     */

    //已经做迁移
    public static ObjectWriteResponse putObject(String bucketName, String objectName,
                                                InputStream inputStream)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient.putObject(
                PutObjectArgs.builder().bucket(bucketName).object(objectName).stream(
                                inputStream, inputStream.available(), -1)
                        .build());
    }

    /**
     * 创建文件夹或目录
     *
     * @param bucketName 存储桶
     * @param objectName 目录路径  （需要/）
     */
    public static ObjectWriteResponse putDirObject(String bucketName, String objectName)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient.putObject(
                PutObjectArgs.builder().bucket(bucketName).object(objectName).stream(
                                new ByteArrayInputStream(new byte[]{}), 0, -1)
                        .build());
    }

    /**
     * 获取文件信息, 如果抛出异常则说明文件不存在
     *
     * @param bucketName bucket名称
     * @param objectName 文件名称
     */
    public static ObjectStat statObject(String bucketName, String objectName)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient
                .statObject(StatObjectArgs.builder().bucket(bucketName).object(objectName).build());
    }

    /**
     * 拷贝文件
     *
     * @param bucketName    bucket名称
     * @param objectName    文件名称
     * @param srcBucketName 目标bucket名称
     * @param srcObjectName 目标文件名称
     */
    //
    public static ObjectWriteResponse copyObject(String bucketName, String objectName,
                                                 String srcBucketName, String srcObjectName)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        return minioClient.copyObject(
                CopyObjectArgs.builder()
                        .source(CopySource.builder().bucket(bucketName).object(objectName).build())
                        .bucket(srcBucketName)
                        .object(srcObjectName)
                        .build());
    }

    /**
     * 删除文件
     *
     * @param bucketName bucket名称
     * @param objectName 文件名称
     */
    //20240424 完成迁移
    public static void removeObject(String bucketName, String objectName)
            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
        minioClient
                .removeObject(RemoveObjectArgs.builder().bucket(bucketName).object(objectName).build());
    }

    /**
     * 下载文件
     * <p>
     * bucketName:
     * 桶名
     *
     * @param fileName: 文件名
     * @param response:
     * @return: void
     * @date : 2020/8/17 0:34
     */
    //20240423 完成迁移
    @SneakyThrows(Exception.class)
    public static void download(String fileName, String realPath, HttpServletResponse response) {
        response.setCharacterEncoding("UTF-8");
        String realFileName = fileName;
        if (StringUtils.isNotBlank(realPath)) {
            realFileName = realFileName.replaceFirst(realPath,"");
        }
        response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(realFileName, "UTF-8"));
        InputStream is = minioClient.getObject(bucketName, fileName);
        IOUtils.copy(is, response.getOutputStream());
        is.close();
    }

    //获取文件流   20240423全局已做迁移
    public static InputStream downloadFile(String fileRelativePath) {
        fileRelativePath = fileRelativePath.replaceAll("\\\\","/");
        try {
            InputStream inputStream = minioClient.getObject(GetObjectArgs.builder()
                    .bucket(bucketName)
                    .object(fileRelativePath)
                    .build());
            return inputStream;
        } catch (Exception e) {
            //LoggerFactory.getLogger(this.getClass()).error("文件下载失败！", e);
        }
        return null;
    }
    /**
     * 获取存储桶策略
     *
     * @param bucketName 存储桶名称
     * @return json
     */

    //暂时不做迁移
//    private JSONObject getBucketPolicy(String bucketName)
//            throws IOException, InvalidKeyException, InvalidResponseException, BucketPolicyTooLargeException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, InsufficientDataException, ErrorResponseException {
//        String bucketPolicy = minioClient
//                .getBucketPolicy(GetBucketPolicyArgs.builder().bucket(bucketName).build());
//        return JSONObject.parseObject(bucketPolicy);
//    }

    /**
     * 断点下载
     *
     * @param bucketName bucket名称
     * @param objectName 文件名称
     * @param offset     起始字节的位置
     * @param length     要读取的长度
     * @return 流
     */
    //暂时不做迁移
//    public InputStream getObject(String bucketName, String objectName, long offset, long length)
//            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
//        return minioClient.getObject(
//                GetObjectArgs.builder().bucket(bucketName).object(objectName).offset(offset).length(length)
//                        .build());
//    }


    //    public static Map<String, String> presignedGetObject(String bucketName, String objectName,
//                                                         Integer expires)
//            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, InvalidExpiresRangeException, ServerException, InternalException, NoSuchAlgorithmException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
//        PostPolicy policy = new PostPolicy(bucketName, objectName,
//                ZonedDateTime.now().plusDays(7));
//        policy.setContentType("image/png");
//        return minioClient.presignedPostPolicy(policy);
//    }

    /**
     * 给presigned URL设置策略
     *
     * @param bucketName 存储桶
     * @param objectName 对象名
     * @param expires    过期策略
     * @return map
     */



    /**
     * 将URLDecoder编码转成UTF8
     *
     * @param str
     * @return
     * @throws UnsupportedEncodingException
     */
//    public static String getUtf8ByURLDecoder(String str) throws UnsupportedEncodingException {
//        String url = str.replaceAll("%(?![0-9a-fA-F]{2})", "%25");
//        return URLDecoder.decode(url, "UTF-8");
//    }



    /**
     * 将文件服务器文件输出到指定目录下文件
     *
     * @param oldPath 文件服务器文件目录
     * @param newPath 输出文件目录
     * @return
     * @throws
     */
//    private int copyFile(String oldPath, String newPath) {
//        try {
//            int byteSum = 0;
//            int byteRead = 0;
//            InputStream inStream = MinioUtils.getObject(MinioUtils.bucketName, oldPath);
//            if(inStream == null){
//                System.out.println(oldPath + "路径文件为空");
//            }else {
//                FileOutputStream fs = new FileOutputStream(newPath);
//                byte[] buffer = new byte[1444];
//                while ( (byteRead = inStream.read(buffer)) != -1) {
//                    byteSum += byteRead; //字节数 文件大小
//                    System.out.println(byteSum);
//                    fs.write(buffer, 0, byteRead);
//                }
//                fs.close();
//            }
//
//            inStream.close();
//        } catch (Exception e) {
//            System.out.println("复制单个文件操作出错");
//            e.printStackTrace();
//        }
//        return 0;
//    }


    // 把地址转为代理地址
//    public static String transURLForProxy(String source, String proxy) {
//        int pos = source.substring(7).indexOf("/");
//        return proxy + "/minio" + source.substring(pos);
//    }


    /**
     * 根据文件前置查询文件
     *
     * @param bucketName bucket名称
     * @param prefix     前缀
     * @param recursive  是否递归查询
     * @return MinioItem 列表
     */
//    public static List<Item> getAllObjectsByPrefix(String bucketName, String prefix,
//                                                   boolean recursive)
//            throws ErrorResponseException, InsufficientDataException, InternalException, InvalidBucketNameException, InvalidKeyException, InvalidResponseException,
//            IOException, NoSuchAlgorithmException, ServerException, XmlParserException {
//        List<Item> list = new ArrayList<Item>();
//        Iterable<Result<Item>> objectsIterator = minioClient.listObjects(
//                ListObjectsArgs.builder().bucket(bucketName).prefix(prefix).recursive(recursive).build());
//        if (objectsIterator != null) {
//            for (Result<Item> o : objectsIterator) {
//                Item item = o.get();
//                list.add(item);
//            }
//        }
//        return list;
//    }


    /**
     * 批量删除文件
     *
     * @param bucketName bucket
     * @param keys       需要删除的文件列表
     * @return
     */

//    public static void removeObjects(String bucketName, List<String> keys) {
//        List<DeleteObject> objects = new LinkedList<DeleteObject>();
//        keys.forEach(s -> {
//            objects.add(new DeleteObject(s));
//            try {
//                removeObject(bucketName, s);
//            } catch (Exception e) {
//                log.error("批量删除失败！error:{}", e);
//            }
//        });
//    }

    /**
     * 获取文件外链
     *
     * @param bucketName bucket名称
     * @param objectName 文件名称
     * @param expires    过期时间 <=7 秒级
     * @return url
     */
//    public static String getPresignedObjectUrl(String bucketName, String objectName,
//                                               Integer expires)
//            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, InvalidExpiresRangeException, ServerException, InternalException, NoSuchAlgorithmException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
//        return minioClient.presignedGetObject(bucketName, objectName, expires);
//    }

    /**
     * 根据bucketName获取信息
     *
     * @param bucketName bucket名称
     */
//    public static Optional<Bucket> getBucket(String bucketName)
//            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
//        return minioClient.listBuckets().stream().filter(b -> b.name().equals(bucketName)).findFirst();
//    }

    /**
     * 根据bucketName删除信息
     *
     * @param bucketName bucket名称
     */
//    public static void removeBucket(String bucketName)
//            throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, InvalidBucketNameException, ErrorResponseException {
//        minioClient.removeBucket(RemoveBucketArgs.builder().bucket(bucketName).build());
//    }

}
